learn probability in python